package com.jacobs.netty.websocketx.benchmarkserver;

/**
 * Created by lichao on 2017/9/12.
 */

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.util.CharsetUtil;

/**
 * Generates the benchmark HTML page which is served at http://localhost:8080/
 */
public final class WebSocketServerBenchmarkPage {

  private static final String NEWLINE = "\r\n";

  public static ByteBuf getContent(String webSocketLocation) {
    return Unpooled.copiedBuffer(
        "<html><head><title>Web Socket Performance Test</title></head>" + NEWLINE +
            "<body>" + NEWLINE +
            "<h2>WebSocket Performance Test</h2>" + NEWLINE +
            "<label>Connection Status:</label>" + NEWLINE +
            "<label id=\"connectionLabel\"></label><br />" + NEWLINE +

            "<form onsubmit=\"return false;\">" + NEWLINE +
            "Message size:" +
            "<input type=\"text\" id=\"messageSize\" value=\"1024\"/><br>" + NEWLINE +
            "Number of messages:" +
            "<input type=\"text\" id=\"nrMessages\" value=\"100000\"/><br>" + NEWLINE +
            "Data Type:" +
            "<input type=\"radio\" name=\"type\" id=\"typeText\" value=\"text\" checked>text" +
            "<input type=\"radio\" name=\"type\" id=\"typeBinary\" value=\"binary\">binary<br>"
            + NEWLINE +
            "Mode:<br>" + NEWLINE +
            "<input type=\"radio\" name=\"mode\" id=\"modeSingle\" value=\"single\" checked>" +
            "Wait for response after each messages<br>" + NEWLINE +
            "<input type=\"radio\" name=\"mode\" id=\"modeAll\" value=\"all\">" +
            "Send all messages and then wait for all responses<br>" + NEWLINE +
            "<input type=\"checkbox\" id=\"verifiyResponses\">Verify responded messages<br>"
            + NEWLINE +
            "<input type=\"button\" value=\"Start Benchmark\"" + NEWLINE +
            "       onclick=\"startBenchmark()\" />" + NEWLINE +
            "<h3>Output</h3>" + NEWLINE +
            "<textarea id=\"output\" style=\"width:500px;height:300px;\"></textarea>" + NEWLINE +
            "<br>" + NEWLINE +
            "<input type=\"button\" value=\"Clear\" onclick=\"clearText()\">" + NEWLINE +
            "</form>" + NEWLINE +

            "<script type=\"text/javascript\">" + NEWLINE +
            "var benchRunning = false;" + NEWLINE +
            "var messageSize = 0;" + NEWLINE +
            "var totalMessages = 0;" + NEWLINE +
            "var rcvdMessages = 0;" + NEWLINE +
            "var isBinary = true;" + NEWLINE +
            "var isSingle = true;" + NEWLINE +
            "var verifiyResponses = false;" + NEWLINE +
            "var benchData = null;" + NEWLINE +
            "var startTime;" + NEWLINE +
            "var endTime;" + NEWLINE +
            "var socket;" + NEWLINE +
            "var output = document.getElementById('output');" + NEWLINE +
            "var connectionLabel = document.getElementById('connectionLabel');" + NEWLINE +
            "if (!window.WebSocket) {" + NEWLINE +
            "  window.WebSocket = window.MozWebSocket;" + NEWLINE +
            '}' + NEWLINE +
            "if (window.WebSocket) {" + NEWLINE +
            "  socket = new WebSocket(\"" + webSocketLocation + "\");" + NEWLINE +
            "  socket.binaryType = 'arraybuffer';" + NEWLINE +
            "  socket.onmessage = function(event) {" + NEWLINE +
            "    if (verifiyResponses) {" + NEWLINE +
            "        if (isBinary) {" + NEWLINE +
            "            if (!(event.data instanceof ArrayBuffer) || " + NEWLINE +
            "                  event.data.byteLength != benchData.byteLength) {" + NEWLINE +
            "                onInvalidResponse(benchData, event.data);" + NEWLINE +
            "                return;" + NEWLINE +
            "            } else {" + NEWLINE +
            "                var v = new Uint8Array(event.data);" + NEWLINE +
            "                for (var j = 0; j < benchData.byteLength; j++) {" + NEWLINE +
            "                    if (v[j] != benchData[j]) {" + NEWLINE +
            "                        onInvalidResponse(benchData, event.data);" + NEWLINE +
            "                        return;" + NEWLINE +
            "                    }" + NEWLINE +
            "                }" + NEWLINE +
            "            }" + NEWLINE +
            "        } else {" + NEWLINE +
            "            if (event.data != benchData) {" + NEWLINE +
            "                onInvalidResponse(benchData, event.data);" + NEWLINE +
            "                return;" + NEWLINE +
            "            }" + NEWLINE +
            "        }" + NEWLINE +
            "    }" + NEWLINE +
            "    rcvdMessages++;" + NEWLINE +
            "    if (rcvdMessages == totalMessages) {" + NEWLINE +
            "        onFinished();" + NEWLINE +
            "    } else if (isSingle) {" + NEWLINE +
            "        socket.send(benchData);" + NEWLINE +
            "    }" + NEWLINE +
            "  };" + NEWLINE +
            "  socket.onopen = function(event) {" + NEWLINE +
            "    connectionLabel.innerHTML = \"Connected\";" + NEWLINE +
            "  };" + NEWLINE +
            "  socket.onclose = function(event) {" + NEWLINE +
            "    benchRunning = false;" + NEWLINE +
            "    connectionLabel.innerHTML = \"Disconnected\";" + NEWLINE +
            "  };" + NEWLINE +
            "} else {" + NEWLINE +
            "  alert(\"Your browser does not support Web Socket.\");" + NEWLINE +
            '}' + NEWLINE +
            NEWLINE +
            "function onInvalidResponse(sent,recvd) {" + NEWLINE +
            "    socket.close();" + NEWLINE +
            "    alert(\"Error: Sent data did not match the received data!\");" + NEWLINE +
            "}" + NEWLINE +
            NEWLINE +
            "function clearText() {" + NEWLINE +
            "    output.value=\"\";" + NEWLINE +
            "}" + NEWLINE +
            NEWLINE +
            "function createBenchData() {" + NEWLINE +
            "    if (isBinary) {" + NEWLINE +
            "        benchData = new Uint8Array(messageSize);" + NEWLINE +
            "        for (var i=0; i < messageSize; i++) {" + NEWLINE +
            "            benchData[i] += Math.floor(Math.random() * 255);" + NEWLINE +
            "        }" + NEWLINE +
            "    } else { " + NEWLINE +
            "        benchData = \"\";" + NEWLINE +
            "        for (var i=0; i < messageSize; i++) {" + NEWLINE +
            "            benchData += String.fromCharCode(Math.floor(Math.random() * (123 - 65) + 65));"
            + NEWLINE +
            "        }" + NEWLINE +
            "    }" + NEWLINE +
            "}" + NEWLINE +
            NEWLINE +
            "function startBenchmark(message) {" + NEWLINE +
            "  if (!window.WebSocket || benchRunning) { return; }" + NEWLINE +
            "  if (socket.readyState == WebSocket.OPEN) {" + NEWLINE +
            "    isBinary = document.getElementById('typeBinary').checked;" + NEWLINE +
            "    isSingle = document.getElementById('modeSingle').checked;" + NEWLINE +
            "    verifiyResponses = document.getElementById('verifiyResponses').checked;" + NEWLINE
            +
            "    messageSize = parseInt(document.getElementById('messageSize').value);" + NEWLINE +
            "    totalMessages = parseInt(document.getElementById('nrMessages').value);" + NEWLINE +
            "    if (isNaN(messageSize) || isNaN(totalMessages)) return;" + NEWLINE +
            "    createBenchData();" + NEWLINE +
            "    output.value = output.value + '\\nStarting Benchmark';" + NEWLINE +
            "    rcvdMessages = 0;" + NEWLINE +
            "    benchRunning = true;" + NEWLINE +
            "    startTime = new Date();" + NEWLINE +
            "    if (isSingle) {" + NEWLINE +
            "        socket.send(benchData);" + NEWLINE +
            "    } else {" + NEWLINE +
            "        for (var i = 0; i < totalMessages; i++) socket.send(benchData);" + NEWLINE +
            "    }" + NEWLINE +
            "  } else {" + NEWLINE +
            "    alert(\"The socket is not open.\");" + NEWLINE +
            "  }" + NEWLINE +
            '}' + NEWLINE +
            NEWLINE +
            "function onFinished() {" + NEWLINE +
            "    endTime = new Date();" + NEWLINE +
            "    var duration = (endTime - startTime) / 1000.0;" + NEWLINE +
            "    output.value = output.value + '\\nTest took: ' + duration + 's';" + NEWLINE +
            "    var messagesPerS = totalMessages / duration;" + NEWLINE +
            "    output.value = output.value + '\\nPerformance: ' + messagesPerS + ' Messages/s';"
            + NEWLINE +
            "    output.value = output.value + ' in each direction';" + NEWLINE +
            "    output.value = output.value + '\\nRound trip: ' + 1000.0/messagesPerS + 'ms';"
            + NEWLINE +
            "    var throughput = messageSize * totalMessages / duration;" + NEWLINE +
            "    var throughputText;" + NEWLINE +
            "    if (isBinary) throughputText = throughput / (1024*1024) + ' MB/s';" + NEWLINE +
            "    else throughputText = throughput / (1000*1000) + ' MChars/s';" + NEWLINE +
            "    output.value = output.value + '\\nThroughput: ' + throughputText;" + NEWLINE +
            "    output.value = output.value + ' in each direction';" + NEWLINE +
            "    benchRunning = false;" + NEWLINE +
            "}" + NEWLINE +
            "</script>" + NEWLINE +
            "</body>" + NEWLINE +
            "</html>" + NEWLINE, CharsetUtil.US_ASCII);
  }

  private WebSocketServerBenchmarkPage() {
    // Unused
  }
}
